# -*- coding: utf-8 -*-
# @Time    : 2021/2/21 22:45
# @Author  : FangQianqian
# @File    : movie.py
# @Software: PyCharm

#import bs4     #网页解析，获取数据
from bs4 import BeautifulSoup     #网页解析，获取数据
import re      #正则表达式，进行文字匹配
import urllib.request, urllib.error      #制定URL，获取网页数据
import xlwt    #进行excel操作
import sqlite3    #进行SQLite数据库操作


#定义主函数
def main():
    baseurl = 'https://movie.douban.com/top250?start='    #基础的url

    #1.爬取网页
    datalist = getData(baseurl)       #调用函数getData(),获取baseurl的数据
    savepath = "豆瓣电影TOP250.xls"  #定义保存数据的路径及文件名

    #2.逐一解析数据

    #3.保存数据
    saveData(datalist,savepath)     #调用函数saveData(),保存数据到savepath定义的文件中

    # askURL("https://movie.douban.com/top250?start=0")

#影片详情链接的规则
findLink = re.compile(r'<a href="(.*?)/">')      #创建正则表达式对象，表示规则,.*?很多个字符出现1次或0次.通过正则表达式获取电影链接
#影片图片的规则
findImgSrc = re.compile(r'<img.*src="(.*?)".*>',re.S)   #re.S,使.匹配包括换行在内的所有字符
#影片片名
findTitle = re.compile(r'<span class="title">(.*?)</span>')
#影片评分
findRating = re.compile(r'<span class="rating_num" property="v:average">(.*)</span>')
#评价人数
findJudge = re.compile(r'<span>(\d*)人评价</span>')
#影片概况
findInq = re.compile(r'<span class="inq">(.*?)</span>')
#影片相关内容
findBd = re.compile(r'<p class="">(.*?)</p>',re.S)


#爬取网页
def getData(baseurl):
    datalist = []

    # 2.逐一解析数据
    for i in range(0,10):    #分析网页的特点，一个页面有25条影片信息，共10个页面。askURL（）函数10次
        url = baseurl + str(i*25)   #“https://movie.douban.com/top250?start=” + “ 每个网页其实值”
        html = askURL(url)     #保存获取到的网页内容

        soup = BeautifulSoup(html,"html.parser")
        for item in soup.find_all('div',class_='item'):          #查找符合要求的字符串，形成列表class_='item'的div
            # print(item)   #测试查看电影item全部信息
            data = []  #保存一部电影的所有信息
            item = str(item)
            # print(item)
            # break


            #影片详情链接
            link = re.findall(findLink,item)[0]     #re库用来通过正则表达式查找指定的字符串
            # print(link)
            data.append(link)     #添加链接

            imgScr = re.findall(findImgSrc,item)
            data.append(imgScr)

            titles = re.findall(findTitle,item)     #片名可能有外文名
            if len(titles) > 1:
                ctitle = titles[0]
                ftitle = titles[1].replace('/','')   #去掉/
                data.append(ctitle)    #中文名
                data.append(ftitle)    #外文名
            else:
                data.append(titles[0])
                data.append(" ")    #外文名留空

            rating = re.findall(findRating,item)
            data.append(rating)


            judgeNum = re.findall(findJudge,item)
            data.append(judgeNum)

            inq = re.findall(findInq,item)
            if len(inq)!=0:
                inq = inq[0].replace("。","")
            else:
                inq = " "
            data.append(inq)


            bd = re.findall(findBd,item)[0]
            bd = re.sub("<br(\s+)?/>(\s+)?","",bd)  # 去掉<br/?
            bd = re.sub("/","",bd)   #替换掉/
            bd = re.sub(" +","",bd)   #去除所有空格
            # bd = bd.strip()    #去掉前后空格
            data.append(bd)

            datalist.append(data)     #把处理好的一部电影信息放入datalist
    # print(datalist)
    return datalist


#得到一个指定url的网页内容
def askURL(url):
    head = {     #模拟浏览器头部信息
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4324.182 Safari/537.36"
    }    #用户代理，告诉访问服务器，访问者是什么类型的浏览器，可以接收什么水平的文件

    request = urllib.request.Request(url,headers=head)     #将伪装信息封装进url进行访问
    html = ""    #定义局部变量
    try:
        response = urllib.request.urlopen(request)     #打开url获取信息
        html = response.read().decode("utf-8")     #读取网页信息，用utf-8解析
        # print(html)

    except urllib.error.URLError as errors:
        if hasattr(errors,'code'):
            print(errors.code)     #打印错误代号
        if hasattr(errors,'reason'):
            print(errors.reason)    #打印错误原因

    return (html)    #将获取到的网页信息返回

#保存数据
def saveData(datalist,savepath):
# 3.保存数据
    book = xlwt.Workbook(encoding="utf-8")  # 创建工作表
    sheet = book.add_sheet("豆瓣电影TOP250",cell_overwrite_ok=True)  # 添加工作页,允许工作页内后写入数据覆盖先前的
    col = ("详情链接","图片链接","中文名","外文名","评分","评价人数","概述","相关信息")
    for i in range(0,8):
        sheet.write(0,i,col[i])     #写入列名
    for j in range(0,250):          #按行写入影片信息
        print('第"%d"条'%(j+1))
        data = datalist[j]
        for i in range(0,8):
            sheet.write(j+1,i,data[i])
    book.save(savepath)



if __name__ == "__main__":     #当程序执行时,程序的开始
#调用函数main()
    main()
    print('保存完成')
